import os
# shutil是高级的文件、文件夹、压缩包处理模块
import shutil

# 定义源文件路径
source_dir = "G:\\image\\天牛\\VOCdevkit\\VOC2007\\YOLOLabels"
# 定义rename_pic函数，传参：源路径和目标路径
def rename_pic(dir_s, dir_t):
    # os.listdir() 返回dir_s包含的文件或文件夹的名字的列表
    pic_list = os.listdir(dir_s)
    i = 1
    # 遍历文件夹每一个文件的名字
    for pic in pic_list:
        # 若文件名中包含. 表示此文件为图片
        if  "." in pic:
            print("pic:", pic)
            # 命名图片名字为数字 str(i)把数值型强转为字符型
            t_name = "img_" + format(i,"0=4") + ".txt"
            # 连接文件夹路径dir_s和图片名称pic
            abs_pic_source_path = os.path.join(dir_s, pic)
            abs_pic_path = os.path.join(dir_t, t_name)
            # 复制文件内容从源路径到目标路径
            shutil.copy(abs_pic_source_path, abs_pic_path)
            i += 1

if __name__=='__main__':
    rename_pic(source_dir, "G:\\image\\txt")
